package com.zzhardy.utils;

import java.io.*;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.util.Map;
import java.util.Objects;

import com.alibaba.fastjson.JSON;
import com.zzhardy.utils.FileUtils;
import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;

/**
 *
 */
public class FreeMakerUtils {
    /**
     *  通过模板生成指定模板文件Str
     * @param templateName 模板名字
     * @param beanMap 模板替换的值
     * @param templateUrl 模板文件路径
     * @return 生成模板的字符串
     */
    public static String templateToStr(String templateName, Map<String, Object> beanMap, String templateUrl) {
        ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader();
        String htmlBody = "";
        StringWriter result = new StringWriter();
        try {
            htmlBody = String.join(System.lineSeparator(),
                                   IOUtils.readLines(new InputStreamReader(Objects.requireNonNull(contextClassLoader.getResourceAsStream(
                                           templateUrl)), StandardCharsets.UTF_8)));
            Configuration config = new Configuration(Configuration.VERSION_2_3_29);
            config.setDefaultEncoding("utf-8");
            config.setClassForTemplateLoading(config.getClass(), "/");
            Template     template      = new Template(templateName, new StringReader(htmlBody), config);
            template.process(beanMap, result);
        } catch (IOException e) {
            e.printStackTrace();
        } catch (TemplateException e) {
            e.printStackTrace();
        }
        return result.toString();
    }
    
    /**
     *  通过模板生成在本地指定模板文件
     * @param beanMap 模板替换的值
     * @param templateUrl 模板文件路径
     * @param fileSaveUrl 模板文件本地保存路径
     */
    public static void templateToFile(Map<String, Object> beanMap, String templateUrl, String fileSaveUrl) {
        try {
            Configuration config = new Configuration(Configuration.VERSION_2_3_29);
            config.setDefaultEncoding("utf-8");
            config.setClassForTemplateLoading(config.getClass(), "/");
            Template      template           = config.getTemplate(templateUrl, "UTF-8");
            File          file               = new File(fileSaveUrl);
            File          parentFile              = new File(file.getParent());
            if (!parentFile.exists()) {
                parentFile.mkdirs();//创建目录
            }
            Writer   out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(fileSaveUrl), "UTF-8"));
            template.process(beanMap, out);
            out.flush();
            out.close();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (TemplateException e) {
            e.printStackTrace();
        }
    }
    
    /**
     * 批量生成代码，可进行代码分层文件夹生成文件夹所有的文件
     * @param generatorName 生成文件的名字前缀
     * @param suffix 生成文件的文件后缀
     * @param templateFolderUrl 批量生成文件的文件夹地址
     * @param fileSaveFolderUrl 批量生成代码的文件夹地址，
     */
    public static void createFileByUrl(String generatorName, String suffix, String templateFolderUrl, String fileSaveFolderUrl) {
        ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader();
        URL          url                = contextClassLoader.getResource(templateFolderUrl);
        File         template           = new File(url.getFile());
        if (template.isDirectory()) {
            File[] files = template.listFiles();
            // 当输入的路径为目录，则进行将该目录下的文件，进行模板生成数据；将目录下的目录进行传递，继续处理相同的业务
            for (int i = 0; i < files.length; i++) {
                File file = files[i];
                if (file.isDirectory()) {
                    String templateFolder = templateFolderUrl + "/" + file.getName();
                    String fileSaveFolder = fileSaveFolderUrl + "/" + file.getName();
                    createFileByUrl(generatorName, suffix, templateFolder, fileSaveFolder);
                } else {
                    // 读取目录下的模板数据，并进行生成模板
                    if (!file.getName().equals("data.json")) {
                        String initDataUrl = url.getFile() + "/data.json";
                        String dataStr = FileUtils.readJsonFile(initDataUrl);
                        Map<String, Object> mapTypes = JSON.parseObject(dataStr);
                        String fileName = (String)mapTypes.get("fileName");
                        if (StringUtils.isNotBlank(fileName)) {
                            String fileNewName = generatorName + fileName;
                            mapTypes.put("fileName", fileNewName);
                            templateToFile(mapTypes, templateFolderUrl + "/" + file.getName(), fileSaveFolderUrl + "/" + fileNewName + suffix);
                        } else {
                            mapTypes.put("fileName", generatorName);
                            templateToFile(mapTypes, templateFolderUrl + "/" + file.getName(), fileSaveFolderUrl + "/" + generatorName + suffix);
                        }
                    }
                }
            }
        }
    }
    
}
